package com.ln.leetcode.train;

import java.util.function.Predicate;

/**
 * 给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。
 * <p>
 * 请你将两个数相加，并以相同形式返回一个表示和的链表。
 * <p>
 * 你可以假设除了数字 0 之外，这两个数都不会以 0 开头。
 * <p>
 * <p>
 * 输入：l1 = [2,4,3], l2 = [5,6,4]
 * 输出：[7,0,8]
 * 解释：342 + 465 = 807.
 * <p>
 * 输入：l1 = [0], l2 = [0]
 * 输出：[0]
 * <p>
 * 输入：l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
 * 输出：[8,9,9,9,0,0,0,1]
 * <p>
 * 提示：
 * <p>
 * 每个链表中的节点数在范围 [1, 100] 内
 * 0 <= Node.val <= 9
 * 题目数据保证列表表示的数字不含前导零
 */
public class TwoNumAdd {

    /**
     * Definition for singly-linked list.
     */
    static public class ListNode {
        int val;
        ListNode next;

        ListNode() {
        }

        ListNode(int val) {
            this.val = val;
        }

        ListNode(int val, ListNode next) {
            this.val = val;
            this.next = next;
        }
    }

    /**
     * 快速生成
     *
     * @param arr 数组
     */
    static ListNode generate(int[] arr) {
        ListNode next = null;
        ListNode current = null;
        for (int i = arr.length - 1; i > -1; i--) {
            current = new ListNode(arr[i], next);
            next = current;
        }
        return current;
    }

    /**
     * 测试结果
     *
     * @param hope 期望结果
     * @param re   运行所得结果
     */
    static String test(String hope, ListNode re) {
        StringBuilder sb = new StringBuilder();
        while (re != null) {
            sb.append(re.val);
            re = re.next;
        }

        Predicate<String> pred = hope::equals;
        return String.format("期望结果: %s, 运算结果: %s，是否正确: %s", hope, sb, pred.test(sb.toString()));
    }

    /**
     * 实现方法 - 递归相加
     */
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        return recursionAdd(l1, l2, 0);
    }


    public ListNode recursionAdd(ListNode l1, ListNode l2, int carry) {
        // 当前两个数相加
        int currentSum = (l1 != null ? l1.val : 0) + (l2 != null ? l2.val : 0) + carry;
        ListNode current = new ListNode(currentSum % 10);
        int nextCarry = currentSum > 9 ? 1 : 0;
        // 递归条件：任一存在下一个节点或者存在进位
        if ((l1 != null && l1.next != null) || (l2 != null && l2.next != null) || nextCarry > 0) {
            ListNode nextNode = recursionAdd(l1 != null ? l1.next : null, l2 != null ? l2.next : null, nextCarry);
            current.next = nextNode;
        }
        return current;
    }



    public static void main(String[] args) {
        TwoNumAdd add = new TwoNumAdd();

        ListNode l1 =  generate(new int[]{2,4,3});
        ListNode r1 =  generate(new int[]{5,6,4});
        System.out.println(test("708", add.addTwoNumbers(l1, r1)));
//
        ListNode l10 =  generate(new int[]{0});
        ListNode r10 =  generate(new int[]{0});
        System.out.println(test("0", add.addTwoNumbers(l10, r10)));
//
        ListNode l11 = generate(new int[]{9, 9, 9, 9, 9, 9, 9}); // 9999999 + 9999 = 10009998
        ListNode r11 = generate(new int[]{9, 9, 9, 9});
        System.out.println(test("89990001", add.addTwoNumbers(l11, r11)));

        ListNode l12 = generate(new int[]{9, 9, 9}); // 999
        ListNode r12 = generate(new int[]{9}); // 9
        // 999 + 9 = 1008
        System.out.println(test("8001", add.addTwoNumbers(l12, r12)));


    }


}
